#pragma once

#include <iostream>
#include <functional>
using namespace std;

void serviceIO_(int sock)
{
    while (true)
    {
        // 这里操作就跟读写文件一样，因为TCP是面向字节流的
        // 因为UDP是面向数据报的，所以要用特殊函数读取，当然TCP也有自己的一套读写接口，后面再讲
        char inbuffer[1024];
        ssize_t n = read(sock, inbuffer, sizeof(inbuffer) - 1);
        if (n > 0)
        {
            // 截止目前我们把读取的内容当成字符串
            inbuffer[n] = 0;
            cout << "recv message: " << inbuffer << endl;

            string outbuffer = inbuffer;
            outbuffer += "[server echo]!";
            write(sock, outbuffer.c_str(), outbuffer.size()); // 等后面讲多路转接再来详谈write()返回值
        }
        else if (n == 0)
        {
            // 代表client退出(类似写端关闭)
            logMessage(NORMAL, "client quit, me too!");
            break;
        }
    }
    close(sock);
}


class Task
{
public:
    using func_t = function<void(int)>;
public:
    Task()
    {
    }

    Task(int sock, func_t func)
        : _sock(sock), _func(func)
    {
    }

    ~Task()
    {
    }

    void operator()()
    {
        _func(_sock);
    }

public:
    int _sock;
    func_t _func;
};
